Fix

quick fix

Signed-off-by: Joshua Pekera <joshuapekera@me.com>