FIX: Forgot to change function name in test.. Oops