Less hacky solution

Signed-off-by: Shea Levy <shea@shealevy.com>