Tests should now pass