Added a test - not working correct