Updated Test case for controllers