Making the code to pass tests once again