Fix Memory Limit

Fix for #1