Working on fix.

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