Make tests green :tada: