Compile Fix

Fix for proper compilation