Fix task controller specs