Fixed the final test :)