Small bug fix.
Corrected missing context in method.