Fixed works controller end error