Updated the test controllers based on david's comment