changes to pass tests