Allow context argument