Fixed tests for last change