Query Workbench: cancel button doesn't return the UI to the user

Trying out the new cancel feature of the workbench, it doesn't return control of the editor to the user until after the query has been executed.

I realize we may not be able to stop the query actually running in the background, but at the moment it appears that the cancel button doesn't function as it should.