cleaning up the mod a little bit
style fix