Forgot to update a variable reference :/