TODO: Fix memory leak