Working on fix

Signed-off-by: Matt Hermes <mathwhiz19@gmail.com>