Fix no context error

[moali87]

Problem: It all depends on if the context is set. If the context isn't set you run into the following below

With the context not set, you no longer get the error