Resolve problem using @jrburke's feedback

Awesome!